WatchCommand.java

/**
 * Copyright (C) 2013 Jakob Külzer (jakob.kuelzer@gmail.com)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.jakusys.jackhammer.cli.upload.command;

import com.google.common.eventbus.EventBus;
import com.google.inject.Inject;
import de.jakusys.jackhammer.cli.path.Path;
import de.jakusys.jackhammer.cli.upload.event.LoggingEventListener;
import de.jakusys.jackhammer.cli.upload.event.UploadingEventListener;
import de.jakusys.jackhammer.cli.upload.handler.factory.DefaultFileHandlerFactory;
import de.jakusys.jackhammer.cli.upload.listener.JackhammerFileAlternationListener;
import de.jakusys.jackhammer.cli.util.PathRelativizer;
import io.airlift.command.Arguments;
import io.airlift.command.Command;
import io.airlift.command.Option;
import org.apache.commons.io.monitor.FileAlterationListener;
import org.apache.commons.io.monitor.FileAlterationMonitor;
import org.apache.commons.io.monitor.FileAlterationObserver;

import javax.jcr.Node;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import java.io.File;

/**
 * @author Jakob Külzer
 */
@Command(name = "watch", description = "Watch a directory for changes and automatically upload them")
public class WatchCommand implements Runnable {

	@Option(name = "--to", description = "Root path in the JCR for uploads, defaults to \"/\"")
	private String rootPath = "";

	@Arguments(title = "directory", description = "Directory to watch for filesystem changes")
	private File directory;

	@Inject
	private Session session;

	@Override
	public void run() {

        if (!directory.exists()) {
            System.out.println("Cannot watch directory " + directory.getPath() + ". The specified path could not be found.");
            return;
        }

		Node node;
		try {

			String tmp = rootPath;
			if (rootPath.startsWith("/"))
				tmp = rootPath.substring(1);

			if ("".equals(tmp)) {
				node = session.getRootNode();
			} else {
				node = session.getRootNode().getNode(tmp);
			}
		} catch (RepositoryException e) {
			throw new RuntimeException("Unable to get node", e);
		}

		final DefaultFileHandlerFactory fileHandlerFactory = new DefaultFileHandlerFactory(new PathRelativizer(directory));

		final EventBus eventBus = new EventBus();
		eventBus.register(new LoggingEventListener(new Path(rootPath)));
		eventBus.register(new UploadingEventListener(fileHandlerFactory, session, node));

		final FileAlterationObserver observer = new FileAlterationObserver(directory);
		final FileAlterationMonitor monitor = new FileAlterationMonitor(1000);

		PathRelativizer pathRelativizer = new PathRelativizer(directory);

		FileAlterationListener listener = new JackhammerFileAlternationListener(eventBus, pathRelativizer);

		observer.addListener(listener);

		monitor.addObserver(observer);
		try {
			System.out.println("Watching " + directory.getCanonicalPath() + " and uploading changes to " + node.getPath());
			monitor.start();

		} catch (Exception e) {
			throw new RuntimeException("Unable to monitor files", e);
		}
	}

}